package javabasics.concurrent.masterworker;

import java.util.HashMap;
import java.util.Map;
import java.util.Queue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedDeque;

/**
 * @Author suxiaoxiao
 * @Date 2019/6/16 0:30
 * @Version 1.0
 */
public class Master {
    /**
     * 任务队列
     * 问题：
     * 1. 为啥用 ConcurrentLinkedDeque ？？？
     * 答： 线程安全的队列
     */
    protected Queue<Object> workQueue = new ConcurrentLinkedDeque<>();
    /**
     * Worker 进程队列
     */
    protected Map<String, Thread> threadMap = new HashMap<>();
    /**
     * 子任务处理结果集
     */
    protected Map<String, Object> resultMap = new ConcurrentHashMap<>();

    /**
     * 是否所有的子任务都结束了
     * @return
     */
    public boolean isComplete() {
        for (Map.Entry<String, Thread> entry : threadMap.entrySet()) {
            /**
             * 线程状态是否是结束状态
             *
             * 线程状态：
             * 新建（new）
             * 可运行（Runnable）
             * 阻塞（block）
             * 无限期等待（waiting）
             * 限期等待（TIMED_WAITING）
             * 结束（terminated）
             */
            if (entry.getValue().getState() != Thread.State.TERMINATED) {
                return false;
            }
        }
        return true;
    }

    /**
     * Master 构造函数
     * @param worker 进程逻辑
     * @param countWorker worker数量
     */
    public Master(Worker worker, int countWorker) {
        worker.setWorkQueue(workQueue);
        worker.setResultMap(resultMap);
        // 初始化几个 worker 线程
        for (int i = 0; i < countWorker; i++) {
            threadMap.put(Integer.toString(i), new Thread(worker, Integer.toString(i)));
        }
    }

    /**
     * 提交一个任务
     * @param job
     */
    public void submit(Object job) {
        workQueue.add(job);
    }

    /**
     * 返回子任务结果集
     * @return
     */
    public Map<String, Object> getResultMap() {
        return resultMap;
    }

    /**
     * 开始运行所有的 Worker 进程，进行处理
     */
    public void execute() {
        for (Map.Entry<String, Thread> entry : threadMap.entrySet()) {
            entry.getValue().start();
        }
    }

}
